package com.leetcode.www;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * Creared with IntelliJ IDEA.
 * Description:给你一个包含 n 个整数的数组nums，判断nums中是否存在三个元素 a，b，c ，使得a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
 *
 * 注意：答案中不可以包含重复的三元组。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/3sum
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * User:yxd
 * Date:2022-03-22
 * Time:17:23
 */
public class Array15 {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ret = new ArrayList<>();
        if(nums == null || nums.length < 3)return ret;

        Arrays.sort(nums);
        for(int i = 0;i < nums.length - 2;i ++){
            if(nums[i] > 0)break;//此时后面就没有能满足和为0的了
            if(i > 0 && nums[i] == nums[i - 1])continue;
            int s = i + 1;
            int e = nums.length - 1;
            while(s < e){
                int sum = nums[i] + nums[s] + nums[e];
                if(sum > 0){
                    while(s < e && nums[e] == nums[--e]);//保证不相等且一直让较大的数向后走直至看sum变化
                }else if(sum < 0){
                    while(s < e && nums[s] == nums[++s]);//保证不相等且一直让较小的数向前走直至看sum变化
                }else{
                    ret.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[s],nums[e])));//添加满足条件的三元组
                    while(s < e && nums[e] == nums[--e]);
                    while(s < e && nums[s] == nums[++s]);
                }
            }
        }
        return ret;
    }
}
